import torch

# 一组任务中的任务数
TASK_NUM = 50

TASK_NUM_GEANT2 = 80

# 训练的episode数量
EPISODE_SIZE = 1000

# 一个episode中的任务组数
GROUP_SIZE = 32

# 设备
DEVICE = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')

# 在预测较好初始权值时运行轮数
INIT_NUM = 50

# 特别需要cpu的情况
DEVICE_CPU = torch.device('cpu')

# 选择图的类型
GRAPH_TAG = ['GBN', 'GEANT2']

# 隐藏层层数
HID_NUM = 50
